import unittest
from login import Login


class TestLogin(unittest.TestCase):

    def test_success(self):
        self.assertEqual('登陆成功', Login("7344", '123'))

    def test_error(self):
        self.assertEqual('登陆失败', Login("123", '123'))
